package com.lap.analysis.file.handler;

import com.lap.analysis.source.entity.ClassInfo;

public class TestFileDirector {

  private final TestFileBuilder builder;

  public TestFileDirector(TestFileBuilder builder) {
    this.builder = builder;
  }

  public String construct(ClassInfo classInfo) {
    builder.builderPackage(classInfo);
    builder.builderImport();
    builder.builderClassHeader(classInfo);
    builder.builderClass(classInfo);

    return builder.getContent();
  }
}
